"""
折线图开发
"""
import json
from pyecharts.charts import Line
from pyecharts.options import *
# 处理数据
file_name_us = "D:/python code/1、Python快速入门（8天零基础入门到精通）/第1-12章资料/可视化案例数据/折线图数据/美国.txt"
f_us = open(file_name_us,"r",encoding="UTF-8")
us_data = f_us.read()
file_name_jp = "D:/python code/1、Python快速入门（8天零基础入门到精通）/第1-12章资料/可视化案例数据/折线图数据/日本.txt"
f_jp = open(file_name_jp,"r",encoding="UTF-8")
jp_data = f_jp.read()
file_name_in = "D:/python code/1、Python快速入门（8天零基础入门到精通）/第1-12章资料/可视化案例数据/折线图数据/印度.txt"
f_in = open(file_name_in,"r",encoding="UTF-8")
in_data = f_in.read()

# 去掉不合JSON规范的开头
us_data = us_data.replace("jsonp_1629344292311_69436(","")
jp_data = jp_data.replace("jsonp_1629350871167_29498(","")
in_data = in_data.replace("jsonp_1629350745930_63180(","")

# 去掉不合JSON规范的结尾
us_data = us_data[:-2]
jp_data = jp_data[:-2]
in_data = in_data[:-2]

# JSON转python字典
us_dict = json.loads(us_data)
jp_dict = json.loads(jp_data)
in_dict = json.loads(in_data)

# 获取trend key
us_trend_data = us_dict['data'][0]['trend']
jp_trend_data = jp_dict['data'][0]['trend']
in_trend_data = in_dict['data'][0]['trend']

# 获取日期数据，用于x轴，取2020年（到314下标结束）
us_x_data  = us_trend_data['updateDate'][:314]
jp_x_data  = jp_trend_data['updateDate'][:314]
in_x_data  = in_trend_data['updateDate'][:314]

# 获取确认数据，用于y轴，取2020年（到314下标结束）
us_y_data = us_trend_data['list'][0]['data'][:314]
jp_y_data = jp_trend_data['list'][0]['data'][:314]
in_y_data = in_trend_data['list'][0]['data'][:314]

# 定义折线图对象
line = Line()
# x轴
line.add_xaxis(xaxis_data=us_x_data)  # x轴数据可以共用
# y轴
# label_opts=LabelOpts(is_show=False)不展示折线上面的数据
line.add_yaxis("美国确诊人数",y_axis=us_y_data,label_opts=LabelOpts(is_show=False))
line.add_yaxis("日本确诊人数",y_axis=jp_y_data,label_opts=LabelOpts(is_show=False))
line.add_yaxis("印度确诊人数",y_axis=in_y_data,label_opts=LabelOpts(is_show=False))
line.set_global_opts(
    # 标题设置
    title_opts = TitleOpts(title="2020年疫情数据折线图",pos_left="center",pos_bottom="1%"),
    # 控制图例
    legend_opts=LegendOpts(is_show=True),
    # 工具箱
    toolbox_opts=ToolboxOpts(is_show=True),
    # 视觉映射
    visualmap_opts=VisualMapOpts(is_show=True)
)
# 生成图标
line.render()
f_us.close()
f_jp.close()
f_in.close()
